#include "udpServer.hpp"
#include <vector>

void Usage(const char* string)
{
    cout << "\n\rUsage: " << string << " port(1024+)\n\n";  
}

string stringHandler(const string& s) 
{
    // 将所有的字符变成大写
    string ret;
    for(auto& e : s) {
        ret += toupper(e);
    }
    return ret;
}

// 检查命令的安全性
bool cmdCheck(const string& s) 
{
    vector<string> dict = {
        "rm", 
        "mv",
        "cp",
        "sudo",
        "yum",
        "unlink",
        "uninstall",
        "top",
        "while"
    };
    for(auto& e : dict) {
        // 如果有关键字中的任意一个，就return false
        if(s.find(e) != string::npos)   return false;
    }
    return true;
}

struct Deleter 
{
    void operator()(FILE* p) 
    {
        if(p != nullptr)    pclose(p);
    }
};

// 执行命令
string exCmd(const string& s) 
{
    cout << "Server get a commad: " << s << '\n'; 
    if(cmdCheck(s) == false)    return "Bad Cmd!";
    // FILE* fp = popen(s.c_str(), "r");
    // unique_ptr<FILE, decltype(&pclose)> fp(popen(s.c_str(), "r"), pclose);
    // unique_ptr<FILE, function<int(FILE*)>> fp(popen(s.c_str(), "r"), pclose);
    // unique_ptr<FILE, function<void(FILE*)>> fp(popen(s.c_str(), "r"), [](FILE* p)->void { if(p) pclose(p); });
    unique_ptr<FILE, Deleter> fp(popen(s.c_str(), "r"));
    if(fp == nullptr) {
        perror("poen: ");
        return "Popen Error!";
    }
    string res;
    while(true) {
        // 将命令输出的字符串读入到buffer中
        char buffer[2*BUFFERSIZE];
        char* p = fgets(buffer, sizeof(buffer), fp.get());
        if(p != nullptr)  res += buffer;
        else break;
    }
    // pclose(fp);
    return res; 
}

// std::string ExcuteCommand(const std::string &cmd)
// {
//     std::cout << "get a request cmd: " << cmd << std::endl;
//     // if(!SafeCheck(cmd)) return "Bad man";
//     FILE *fp = popen(cmd.c_str(), "r");
//     if(nullptr == fp)
//     {
//         perror("popen");
//         return "error";
//     }
//     std::string result;
//     char buffer[4096];
//     while(true)
//     {
//         char *ok = fgets(buffer, sizeof(buffer), fp);
//         if(ok == nullptr) break;
//         result += buffer;
//     }
//     pclose(fp);
//     return result;
// }

string stringHandler2(const string& s, const string& ip, const uint16_t& port) 
{
    cout << "From [" << ip << ' ' << port << "]# " << s << endl;
    string ret;
    // 将所有的字符变成大写
    for(auto& e : s) {
        ret += toupper(e);
    }
    return ret;
}

int main(int argc, char* argv[])
{
    if(argc != 2) {
        Usage(argv[0]);
    } else {
        uint16_t port = stoi(argv[1]);
        //  使用智能指针管理资源
        unique_ptr<UdpServer> svr(new UdpServer(port));
        svr->init();
        // svr->run2(stringHandler2);
        svr->run3();
    }
    return 0;
}